package leetcode._06_回溯;

import org.junit.Test;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.StringJoiner;

/**
 * @author pppppp
 * @date 2022/3/1 19:11
 * 有效 IP 地址 正好由四个整数（每个整数位于 0 到 255 之间组成，且不能含有前导 0），整数之间用 '.' 分隔。
 * <p>
 * 例如："0.1.2.201" 和 "192.168.1.1" 是 有效 IP 地址，
 * 但是 "0.011.255.245"、"192.168.1.312" 和 "192.168@1.1" 是 无效 IP 地址。
 * 给定一个只包含数字的字符串 s ，用以表示一个 IP 地址，返回所有可能的有效 IP 地址，
 * 这些地址可以通过在 s 中插入 '.' 来形成。你 不能 重新排序或删除 s 中的任何数字。你可以按 任何 顺序返回答案。
 * <p>
 * 示例 1：
 * <p>
 * 输入：s = "25525511135"
 * 输出：["255.255.11.135","255.255.111.35"]
 * 示例 2：
 * <p>
 * 输入：s = "0000"
 * 输出：["0.0.0.0"]
 * 示例 3：
 * <p>
 * 输入：s = "101023"
 * 输出：["1.0.10.23","1.0.102.3","10.1.0.23","10.10.2.3","101.0.2.3"]
 */
public class _93_复原IP地址 {

    @Test
    public void T_review() {
        // List<String> list = restoreIpAddresses_review("0279245587303");
        List<String> list = restoreIpAddresses_review2("101023");
        System.out.println();
    }
    private List<String> restoreIpAddresses_review2(String s) {
        if(s.length() > 12 || s.length() < 4){
            return res;
        }
        dfs_review2(0,s);
        return res;
    }

    private void dfs_review2(int index, String s) {

        if(path.size() == 3){
            if(isV(s.substring(index))){
                res.add(getStr(path) + "." + s.substring(index));
            }
            return;
        }

        if(index == s.length()){
            return;
        }
        for (int i = index; i < index + 3 && i < s.length(); i++) {
            String num = s.substring(index, i + 1);
            if(isV(num)){
                path.add(num);
                dfs_review2(i + 1,s);
                path.removeLast();
            }
        }
    }

    private boolean isV(String num) {
        if(num.length() == 0){
            return false;
        }
        if(num.equals("0")){
            return true;
        }
        if(num.startsWith("0")){
            return false;
        }
        return Integer.valueOf(num) < 255;
    }

    private String getStr(LinkedList<String> path) {
        StringJoiner stringJoiner = new StringJoiner(".");
        for (String s : path) {
            stringJoiner.add(s);
        }
        return String.valueOf(stringJoiner);
    }


    private List<String> restoreIpAddresses_review(String s) {
        if (s.length() == 0 || s.length() > 12) {
            return res;
        }
        dfs_review(0, s);
        return res;
    }

    private void dfs_review(int i, String s) {
        if (path.size() == 3) {
            String sLastOne = s.substring(i);
            if (isVaild(sLastOne)) {

                StringBuilder sb = new StringBuilder();
                for (String s1 : path) {
                    sb.append(s1 + ".");
                }
                sb.append(sLastOne);
                res.add(new String(sb));
            }
            return;
        }
        for (int j = i; j < i + 3 && j < s.length(); j++) {
            if (isVaild(s.substring(i, j + 1))) {
                path.add(s.substring(i, j + 1));
                dfs_review(j + 1, s);
                path.removeLast();
            } else {
                break;
            }
        }
    }

    private boolean isVaild(String s) {

        if (s.length() == 0 || s.length() > 3 || Integer.valueOf(s) > 255 || (s.startsWith("0") && s.length() != 1)) {
            return false;
        }

        return true;
    }


    @Test
    public void T_() {
        List<String> list = restoreIpAddresses2("0279245587303");
        System.out.println();
    }

    ArrayList<String> res = new ArrayList<>();
    LinkedList<String> path = new LinkedList<>();

    public List<String> restoreIpAddresses2(String s) {
        if (s.length() == 0 || s.length() > 12) {
            return res;
        }
        dfs(s);
        return res;
    }

    private void dfs(String s) {
        if (path.size() == 3) {
            if (check(s)) {
                StringBuilder sb = new StringBuilder();
                for (String s1 : path) {
                    sb.append(s1 + ".");
                }
                sb.append(s);
                res.add(sb.toString());
            }
            return;
        }
        /*剪枝*/  /*有点多余*/
        int maxSpace = (4 - path.size()) * 3;
        if (s.length() > maxSpace || (s.length() == maxSpace && !check(s.substring(0, 3)))) {
            return;
        }
        for (int i = 1; i <= 3 && i <= s.length() - 1; i++) {
            if (check(s.substring(0, i))) {
                path.add(s.substring(0, i));
                dfs(s.substring(i));
                path.removeLast();
            }
        }
    }

    @Test
    public void T_0(){
        // String s = "25525511135";
        String s = "101023";
        List<String> strings = restoreIpAddresses(s);
        System.out.println();
    }
    /*暴力破解 4个 for*/
    public List<String> restoreIpAddresses(String s) {
        List<String> res = new ArrayList<>();
        if (s.length() == 0 || s.length() > 12) {
            return res;
        }

        int len = s.length();
        // char[] ch = s.toCharArray();
        for (int l = 1; l <= 3; l++) {
            for (int m = l + 1; m - l <= 3; m++) {
                for (int r = len - 1; r >= len - 3 && r > m; r--) {
                    if (check(s.substring(0, l))
                            && check(s.substring(l, m))
                            && check(s.substring(m, r))
                            && check(s.substring(r))) {
                        res.add(s.substring(0, l) + "." + s.substring(l, m) + "." + s.substring(m, r) + "." + s.substring(r));
                    }
                }
            }

        }
        return res;
    }

    private boolean check(String s) {
        if (s.length() > 3) {
            return false;
        }
        if (s.startsWith("0")) {
            return s.length() == 1;
        }
        if (Integer.valueOf(s) >= 0 && Integer.valueOf(s) <= 255) {
            return true;
        }
        return false;
    }

}
